python dsa question